public class UserContext {
    // 使用ThreadLocal存储当前登录用户（线程安全）
    private static ThreadLocal<User> currentUser = new ThreadLocal<>();

    // 设置当前用户
    public static void setCurrentUser(User user) {
        currentUser.set(user);
    }

    // 获取当前用户
    public static User getCurrentUser() {
        return currentUser.get();
    }

    // 清除当前用户（退出登录时调用）
    public static void clear() {
        currentUser.remove();
    }
}
    